import openpyxl
import invite2
# 引入sys模块，用来读取参数
import sys
# 判断参数，如果参数小于2，则提示用户输入文件名并退出程序
# Python程序的第1个参数是Python文件名，所以第2个参数才是文件名
if len(sys.argv) < 2:
    print('运行时必须输入名单文件名，如names.xlsx')
    sys.exit()
# 读取文件名
filename = sys.argv[1]
excel_file = openpyxl.load_workbook(filename)
worksheet = excel_file.active
for index, row in enumerate(worksheet.rows):
    if index > 0:
        name = row[0].value
        # 根据名字所在的行数决定要保存在那个子目录下。
        # index除以1000，然后转成整数，就让0到999的index转换后都成了0，1000到1999都变成了1，以此类推。
        subfolder = int(index/1000)
        # 给invite函数传入由letters以及子目录拼成的目录
        invite2.invite(name, f'letters/{subfolder}')
